package com.samuel.algorithm.array;

import java.util.Arrays;

/**
 * <h1>合并两个有序数组</h1>
 * 给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。
 * 请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。<br\>
 * 注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。
 * 为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n
 */
class MergeSolution {


    /**
     * 解法1：直接合并后排序
     * 把nums2直接放到nums1的尾部，然后在对nums1排序
     */
    public void merge1(int[] nums1, int m, int[] nums2, int n) {
        for (int i = 0; i < n; i++) {
            nums1[m + i] = nums2[i];
        }
        Arrays.sort(nums1);
    }

    /**
     * 解法2：双指针法
     * 借助临时数组把nums1和nums2排序后放入临时数组中，然后临时数组赋值给nums1
     */
    public void merge2(int[] nums1, int m, int[] nums2, int n) {
        int[] temp = new int[m + n];
        int p1 = 0;
        int p2 = 0;
        int tp = 0;
        while (p1 < m || p2 < n) {
            if (p1 >= m) {
                temp[tp] = nums2[p2];
                p2++;
            } else if (p2 >= n) {
                temp[tp] = nums1[p1];
                p1++;
            } else if (nums1[p1] < nums2[p2]) {
                temp[tp] = nums1[p1];
                p1++;
            } else {
                temp[tp] = nums2[p2];
                p2++;
            }
            tp++;
        }
        for (int i = 0; i < temp.length; i++) {
            nums1[i] = temp[i];
        }
    }

    /**
     * 解法3：逆向双指针
     * 相较于解法2不需要借助临时数组，节省空间
     */
    public void merge3(int[] nums1, int m, int[] nums2, int n) {
        int p1 = m-1, p2 = n-1;
        int temp = m + n -1;
        while (temp >= 0) {
            if (p1< 0 && p2 < 0) {
                return;
            }
            if (p1 < 0) {
                nums1[temp] = nums2[p2];
                p2--;
            } else if (p2 < 0) {
                nums1[temp] = nums1[p1];
                p1--;
            } else if (nums1[p1] > nums2[p2]) {
                nums1[temp] = nums1[p1];
                p1--;
            } else {
                nums1[temp] = nums2[p2];
                p2--;
            }
            temp--;
        }
    }

    public static void main(String[] args) {
        int[] n1 = {1, 2, 3, 0, 0, 0};
        int[] n2 = {2, 5, 6};
        MergeSolution mergeSolution = new MergeSolution();
        mergeSolution.merge3(n1, 3, n2, 3);
    }
}